Train yolov8 model for classification